import numpy as np
import matplotlib.pyplot as plt

data = np.arange(0, np.pi * 2, 0.01)
fig = plt.figure(figsize=(9, 7), dpi=90)  # 确定画布大小

# 绘制第一幅子图
ax1 = fig.add_subplot(1, 2, 1)
plt.title('lines example')
plt.xlabel('X')
plt.ylabel('Y')
plt.xlim(0, 1)
plt.ylim(0, 1)
plt.xticks([0, 0.2, 0.4, 0.6, 0.8, 1])
plt.yticks([0, 0.2, 0.4, 0.6, 0.8, 1])
plt.plot(data, data ** 2)
plt.plot(data, data ** 3)
plt.legend(['y=x^2', 'y=x ^3'], loc='best', frameon=False)  # 去掉图例边框,推荐使用

# 绘制第二幅子图
ax2 = fig.add_subplot(1, 2, 2)
plt.title('sin-cos')
plt.xlabel('X')
plt.ylabel('Y')
plt.xlim(0, np.pi * 2)
plt.ylim(-1, 1)
plt.xticks([0, np.pi / 2, np.pi, np.pi * 3 / 2, np.pi * 2])
plt.yticks([-1, -0.5, 0, 0.5, 1])
plt.plot(data, np.sin(data))
plt.plot(data, np.cos(data))
# plt.legend(['sin', 'cos'], loc='best', edgecolor='blue')  # 设置图例边框颜色
plt.legend(['sin', 'cos'], loc='best', facecolor='green')  # 设置图例背景颜色,若无边框,参数无效

# 保存图片,transparent=True指定图片透明度,pad_inches=0去除所有白边
fig.savefig("pic/ex6.png", format='png', transparent=True, dpi=300, pad_inches=0)
plt.show()  # 显示图片
